import page
from base.base import Base
from tool.get_log import GetLogger

log = GetLogger().get_log()


class PageWdOther(Base):

    def pagewdother_click_update_tiku(self):
        self.base_click_element(page.wd_update_tiku)

    def pagewdother_check_tiku_toast(self):
        if self.base_check_toast("已是最新啦") or self.base_check_toast("更新完成"):
            return True
        else:
            return False

    def pagewdother_click_share_app(self):
        self.base_click_element(page.wd_share_app)

    def pagewdother_share_app_to_QQ(self):
        self.base_click_element(page.wd_share_app)
        self.base_click_element(page.wd_share_app_to_QQ)
        try:
            self.base_click_element(page.right_next_step)
            self.base_click_element(page.wd_share_app_agree_btn)
            self.base_click_element(page.wd_share_app)
            self.base_click_element(page.wd_share_app_to_QQ)
        except:
            pass

    def pagewdother_return_to_wdpage_keycode(self):
        for a in range(10):
            log.info(f'正在返回我的页面，第{a + 1}次尝试')
            if self.base_if_element_exist(page.wd_tab):
                break
            else:
                self.driver.press_keycode(4)

    def pagewdother_if_share_success(self):
        if self.base_get_current_package_name() == 'com.tencent.mobileqq':
            return True
        else:
            return False
